from experiment.models import Experiment, Race, SubjectCode, Subject, Phone, Email, Contact, CompensationType, CompensationSource, Location, SubjectSession, SubjectSessionFile
from django.contrib import admin
from reversion.admin import VersionAdmin

class ExperimentAdmin(VersionAdmin):
    """Add Documentation"""

class RaceAdmin(VersionAdmin):
    """Add Documentation"""

class SubjectCodeAdmin(VersionAdmin):
    """Add Documentation"""

class SubjectAdmin(VersionAdmin):
    """Add Documentation"""
    fields = ('experiment', 'gender', 'handedness', 'ethnicity', 'race', 'other', 'codes', 'notes')

class PhoneAdmin(VersionAdmin):
    """Add Documentation"""

class EmailAdmin(VersionAdmin):
    """Add Documentation"""

class ContactAdmin(VersionAdmin):
    """Add Documentation"""

class CompensationTypeAdmin(VersionAdmin):
    """Add Documentation"""

class CompensationSourceAdmin(VersionAdmin):
    """Add Documentation"""

class LocationAdmin(VersionAdmin):
    """Add Documentation"""

class SubjectSessionAdmin(VersionAdmin):
    """Add Documentation"""
    fields = ('subject', 'session', 'age', 'date_tested', 'compensation', 'compensation_type', 'compensation_source', 'location', 'consenter', 'researchers', 'consented', 'copy_consent', 'completed', 'adverse_event', 'notes')


class SubjectSessionFileAdmin(VersionAdmin):
    """Add Documentation"""


admin.site.register(Experiment, ExperimentAdmin)
admin.site.register(Race, RaceAdmin)
admin.site.register(SubjectCode, SubjectCodeAdmin)
admin.site.register(Subject, SubjectAdmin)
admin.site.register(Phone, PhoneAdmin)
admin.site.register(Email, EmailAdmin)
admin.site.register(Contact, ContactAdmin)
admin.site.register(CompensationType, CompensationTypeAdmin)
admin.site.register(CompensationSource, CompensationSourceAdmin)
admin.site.register(Location, LocationAdmin)
admin.site.register(SubjectSession, SubjectSessionAdmin)
admin.site.register(SubjectSessionFile, SubjectSessionFileAdmin)

